# SPDX-FileCopyrightText: Copyright (c) 2024, NVIDIA CORPORATION & AFFILIATES.
# All rights reserved.
# SPDX-License-Identifier: Apache-2.0

from .pipeline_builders import setup_ingestion_pipeline
from .stage_builders import (
    add_sink_stage,
    add_source_stage,
    add_submitted_job_counter_stage,
    add_metadata_injector_stage,
    add_pdf_extractor_stage,
    add_image_extractor_stage,
    add_docx_extractor_stage,
    add_pptx_extractor_stage,
    add_image_dedup_stage,
    add_image_filter_stage,
    add_table_extractor_stage,
    add_chart_extractor_stage,
    add_image_caption_stage,
    add_nemo_splitter_stage,
    add_embed_extractions_stage,
    add_embedding_storage_stage,
    add_image_storage_stage,
    add_vdb_task_sink_stage,
)

__all__ = [
    "setup_ingestion_pipeline",
    "add_sink_stage",
    "add_source_stage",
    "add_submitted_job_counter_stage",
    "add_metadata_injector_stage",
    "add_pdf_extractor_stage",
    "add_image_extractor_stage",
    "add_docx_extractor_stage",
    "add_pptx_extractor_stage",
    "add_image_dedup_stage",
    "add_image_filter_stage",
    "add_table_extractor_stage",
    "add_chart_extractor_stage",
    "add_image_caption_stage",
    "add_nemo_splitter_stage",
    "add_embed_extractions_stage",
    "add_embedding_storage_stage",
    "add_image_storage_stage",
    "add_vdb_task_sink_stage",
]
